package com.example.mq.mqserver.core;

import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;
import lombok.Getter;
import lombok.Setter;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 这个类表示一个存储消息的队列
 * MSG =》 Message
 */
@Data
public class MSGQueue {
    //表示队列的身份表示（唯一）
    private String name;
    //表示队列是否持久化
    private boolean durable=false;
    //这个属性为true，表示这个队列只能被一个消费者使用（别人用不了），如果是false则是大家都能使用
    //这个独占功能，也是把字段列在这里，具体的功能先不实现
    private boolean exclusive=false;
    //为true表示没有人使用就会自动删除
    //这个属性暂时列在这里，后续代码中并没有真的实现这个自动删除功能
    private boolean autoDelete=false;
    //arguments表示的是创建交换机是指定的一些额外的参数选项，后续代码中没有实现
    private Map<String,Object> arguments=new HashMap<>();
    //当前队列都有哪些消费者订阅了
    private List<ConsumerEnv> consumerEnvList=new ArrayList<>();
    //记录当前取到了第几个消费者，方便实现轮询策略
    //原子类型的，防止多线程情况下被修改
    private AtomicInteger consumerSeq=new AtomicInteger(0);

    //添加一个新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv){
        synchronized (this) {
            consumerEnvList.add(consumerEnv);
        }
    }
    //订阅者的删除先不考虑
    //挑选一个订阅者，用来处理当前的消息（按照轮询的方式）
    public ConsumerEnv chooseConsumer(){
        if (consumerEnvList.size() == 0) {
            // 该队列没有人订阅的
            return null;
        }
        // 计算一下当前要取的元素的下标.
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }

    public String  getArguments(){
        //是把当前的 arguments参数，从 map转成 String（JSON）
        ObjectMapper objectMapper=new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        //如果代码真的异常了，返回一个空的 json对象
        return "{}";
    }

    //这个方法，是从数据库读取数据之后，构造Exchange对象，会自动调用到
    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper=new ObjectMapper();
        //把JSON字符串，转为map
        //如果是个简单类型，直接使用对应类型的类对象即可
        //如果是集合类这样的复杂类型，可以使用 TypeReference匿名内部类对象，来描述复杂类型的具体信息（通过泛型参数来描述的）
        try {
            this.arguments=objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String,Object>>(){});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }
    //这里针对arguments，再提供一组getter和setter，用来去更方便的获取/设置键值对
    //在Java内部使用，比如测试的时候使用
    public Object getArguments(String key){
        return arguments.get(key);
    }

    public void setArguments(String key,Object value){
        arguments.put(key,value);
    }
    public void setArguments(Map<String,Object> arguments){
        this.arguments=arguments;
    }
}
